//    CANAPE Network Testing Tool
//    Copyright (C) 2014 Context Information Security
//
//    This program is free software: you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation, either version 3 of the License, or
//    (at your option) any later version.
//
//    This program is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License
//    along with this program.  If not, see <http://www.gnu.org/licenses/>.

using Microsoft.CSharp;
using System.IO;
using System;
using System.CodeDom;

namespace CANAPE.Scripting.DotNet
{
    [ScriptEngine("csharp", "C#")]
    internal class CSharpScriptEngine : DotNetScriptEngine
    {
        public CSharpScriptEngine(bool enableDebug)
            : base(new CSharpCodeProvider(), enableDebug ? null : "/optimize")
        {
        }

        /// <summary>
        /// Overridden code generation, used to clear up the code slightly to reduce size
        /// This is mainly because this is only used for parser generation where size is 
        /// an important factor
        /// </summary>
        /// <param name="unit">The code unit</param>
        /// <returns>The generated code</returns>
        public override string GenerateCode(CodeCompileUnit unit)
        {
            string s = base.GenerateCode(unit);

            StringReader reader = new StringReader(s);
            string line = reader.ReadLine();

            // Strip off initial comments
            while (line != null)
            {
                if (!String.IsNullOrWhiteSpace(line) && !line.StartsWith("//"))
                {
                    s = line + Environment.NewLine + reader.ReadToEnd();
                    break;
                }

                line = reader.ReadLine();
            }

            return s;
        }        
    }
}
